import os
import yaml


class Config:
    def __init__(self):
        pass

    def load_config(self, config_file_path):
        """
        从 YAML 文件加载配置数据。

        :param config_file_path: 配置文件的路径
        :return: 加载的配置数据字典
        """
        with open(config_file_path, 'r', encoding='utf-8') as file:
            return yaml.safe_load(file)

    def load_test_data(self, data_file_path):
        """
        从 YAML 文件加载测试数据。

        :param data_file_path: 测试数据文件的路径
        :return: 加载的测试数据列表
        """
        with open(data_file_path, 'r', encoding='utf-8') as file:
            data = yaml.safe_load(file)
            if isinstance(data, list):
                return data  # 如果 YAML 文件中的数据是一个列表，则直接返回
            else:
                raise ValueError("YAML 文件内容应该是一个列表")


# 创建配置实例
config = Config()
